import matplotlib.pyplot as plt
import pandas as pd

"""
热力图：
    语法格式：
        plt.imshow(x)
    
    参数说明：
        x:数据
"""
x = [
    [1, 2],
    [3, 4],
    [5, 6],
    [7, 8],
    [9, 10]
]

plt.imshow(x)
# 显示颜色条
plt.colorbar()
plt.show()

# 使用热力图
df = pd.read_excel('libs/成绩表.xlsx')

# 解决中文乱码
plt.rcParams['font.sans-serif'] = ['SimHei']
x = df.loc[:, '数学':'英语'].values
print(x)

# 设置x轴的坐标轴的刻度
plt.xticks(range(3), ['数学', '语文', '英语'])
plt.yticks(range(10),df['姓名'])

# 绘制热力图
plt.imshow(x)

# 显示颜色条
plt.colorbar()

# 设置标题
plt.title('学生成绩热力图')
plt.show()
